import ollama

# import ollama
# # 1. 获取prompt提示词
# prompt = """
# 请为以下功能生成一段Python代码：
# 求两个数的最大公约数
# """
# response = ollama.chat(model='qwen2:0.5b',
#                        messages=[{'role': 'user', 'content': prompt}, ])
# # 2. 调用模型，回答问题
# result = response['message']['content']
# # 3. 打印回答
# print(result)

# def gcd(a, b):
#     while b != 0:
#         a, b = b, a % b
#     return a
#
# # 使用示例：
# a = int(input("请输入第一个整数: "))
# b = int(input("请输入第二个整数: "))
#
# print("最大公约数是:", gcd(a, b))
# 4. 获取prompt提示词
prompt = """
你需要对用户的反馈进行原因分类。
分类包括：价格过高、售后支持不足、产品使用体验不佳、其他。
回答格式为：分类结果：xx。
用户的问题是：产品太重,根本提不动,颜色也不好看。
"""
response = ollama.chat(model='deepseek-r1:1.5b',
                       messages=[{'role': 'user', 'content': prompt}, ])
# 5. 调用模型，回答问题
result = response['message']['content']
# 6. 打印回答
print(result)
























